Automated Machine Learning with AutoKeras by Luis Sobrecueva

Author:Luis Sobrecueva
Language: eng
Format: epub
Publisher: Packt Publishing Pvt Ltd
Published: 2021-04-22T00:00:00+00:00

Creating and fine-tuning a powerful image classifier

We will now use the AutoKeras ImageClassifier class to find the best classification model. Just for this little example, we set max_trials (the maximum number of different Keras models to try) to 2, and we do not set the epochs parameter so that it will use an adaptive number of epochs automatically. For real use, it is recommended to set a large number of trials. The code is shown here:

clf = ak.ImageClassifier(max_trials=2)

Let's run the training to search for the optimal classifier for the CIFAR-10 training dataset, as follows:

clf.fit(x_train, y_train)

Here is the output:

Download

Copyright Disclaimer:
This site does not store any files on its server. We only index and link to content provided by other sites. Please contact the content providers to delete copyright contents if any and email us, we'll remove relevant links or contents immediately.